package com.zou.practice.leetcode;

import com.alibaba.fastjson.JSON;

import java.util.HashMap;

/**
 * 给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
 * 你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/two-sum
 *
 * @author zouyaowen
 * @date 2020-08-31
 */
public class TwoSum01 {

    public static void main(String[] args) {
        int[] nums = {2, 7, 11, 15};
        int target = 9;
        int[] res = twoSum_Solution2(nums, target);
        System.out.println(JSON.toJSONString(res));
    }

    /**
     * 两数之和-> 数组中两个元素值之和等于目标值，返回数组两个数据的索引
     * 暴力方案  时间复杂度O(n方）
     *
     * @param nums   数组参数
     * @param target 目标值
     * @return 索引结果
     */
    public static int[] twoSum_Solution1(int[] nums, int target) {
        if (nums == null || nums.length < 2) {
            return new int[]{-1, -1};
        }
        int[] indexArr = new int[2];
        for (int i = 0; i < nums.length; i++) {
            for (int j = 1; j < nums.length; j++) {
                if (i != j && nums[i] + nums[j] == target) {
                    indexArr[0] = i;
                    indexArr[1] = j;
                    return indexArr;
                }
            }
        }
        return new int[]{-1, -1};
    }

    /**
     * 两数之和-> 数组中两个元素值之和等于目标值，返回数组两个数据的索引
     * 使用hashMap  时间复杂度O(n)
     *
     * @param nums   数组参数
     * @param target 目标值
     * @return 索引结果
     */
    public static int[] twoSum_Solution2(int[] nums, int target) {
        if (nums == null || nums.length < 2) {
            return new int[]{-1, -1};
        }
        HashMap<Integer, Integer> dataMap = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            if (dataMap.containsKey(target - nums[i])) {
                return new int[]{dataMap.get(target - nums[i]), i};
            }
            dataMap.put(nums[i], i);
        }
        return new int[]{-1, -1};
    }
}
